package top.byteio.common.core.utils;

import cn.hutool.extra.spring.SpringUtil;
import lombok.Getter;
import org.springframework.beans.BeansException;
import org.springframework.boot.autoconfigure.thread.Threading;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ApplicationEvent;
import org.springframework.core.env.Environment;
import org.springframework.lang.NonNull;
import org.springframework.stereotype.Component;

import java.util.Map;

/**
 * Spring工具类
 *
 * @author lerryxia
 */
@Component
public class SpringUtils extends SpringUtil implements ApplicationContextAware {

    @Getter
    private static ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(@NonNull ApplicationContext applicationContext) throws BeansException {
        if (SpringUtils.applicationContext == null) {
            SpringUtils.applicationContext = applicationContext;
        }
    }

    /**
     * 通过name获取Bean
     *
     * @param name name
     * @return java.lang.Object
     * @author lerryxia
     * @since 2023/6/21 15:30
     */
    @SuppressWarnings("unchecked")
    public static <T> T getBean(String name) {
        return (T) getApplicationContext().getBean(name);
    }

    /**
     * 通过class获取Bean
     *
     * @param clazz class
     * @return T
     * @author lerryxia
     * @since 2023/6/21 15:28
     */
    public static <T> T getBean(Class<T> clazz) {
        return getApplicationContext().getBean(clazz);
    }


    /**
     * 通过name,以及Clazz返回指定的Bean
     *
     * @param name  name
     * @param clazz class
     * @return T
     * @author lerryxia
     * @since 2023/6/21 15:27
     */
    public static <T> T getBean(String name, Class<T> clazz) {
        return getApplicationContext().getBean(name, clazz);
    }
    
    /**
     * 获取Environment
     *
     * @return Environment
     */
    public static Environment getEnvironment() {
        return getApplicationContext().getEnvironment();
    }
    
    /**
     * 获取配置文件中的属性值
     *
     * @param key 配置键
     * @return 属性值
     */
    public static String getProperty(String key) {
        return getEnvironment().getProperty(key);
    }
    
    /**
     * 获取配置文件中的属性值
     *
     * @param key 配置键
     * @param defaultValue 默认值
     * @return 属性值
     */
    public static String getProperty(String key, String defaultValue) {
        return getEnvironment().getProperty(key, defaultValue);
    }
    
    /**
     * 获取配置文件中的属性值
     *
     * @param key 配置键
     * @param targetType 目标类型
     * @return 属性值
     */
    public static <T> T getProperty(String key, Class<T> targetType) {
        return getEnvironment().getProperty(key, targetType);
    }
    
    /**
     * 获取配置文件中的属性值
     *
     * @param key 配置键
     * @param targetType 目标类型
     * @param defaultValue 默认值
     * @return 属性值
     */
    public static <T> T getProperty(String key, Class<T> targetType, T defaultValue) {
        return getEnvironment().getProperty(key, targetType, defaultValue);
    }
    
    /**
     * 获取一组指定类型的Bean
     *
     * @param type Bean类型
     * @param <T> Bean类型
     * @return Bean映射
     */
    public static <T> Map<String, T> getBeansOfType(Class<T> type) {
        return getApplicationContext().getBeansOfType(type);
    }
    
    /**
     * 发布事件
     *
     * @param event 事件
     */
    public static void publishEvent(ApplicationEvent event) {
        getApplicationContext().publishEvent(event);
    }
    
    /**
     * 发布事件
     *
     * @param event 事件
     */
    public static void publishEvent(Object event) {
        getApplicationContext().publishEvent(event);
    }
    
    /**
     * 获取当前应用名称
     *
     * @return 应用名称
     */
    public static String getApplicationName() {
        return getProperty("spring.application.name", "application");
    }
    
    /**
     * 判断Bean是否存在
     *
     * @param beanName Bean名称
     * @return 是否存在
     */
    public static boolean containsBean(String beanName) {
        return getApplicationContext().containsBean(beanName);
    }
    
    /**
     * 判断Bean是否为单例
     *
     * @param beanName Bean名称
     * @return 是否为单例
     */
    public static boolean isSingleton(String beanName) {
        return getApplicationContext().isSingleton(beanName);
    }
    
    /**
     * 获取Bean的类型
     *
     * @param beanName Bean名称
     * @return Bean类型
     */
    public static Class<?> getType(String beanName) {
        return getApplicationContext().getType(beanName);
    }

    /**
     * 判断当前环境是否为虚拟环境
     *
     * @return boolean 返回true表示当前环境是虚拟环境，false表示不是虚拟环境
     */
    public static boolean isVirtual() {
        return Threading.VIRTUAL.isActive(getBean(Environment.class));
    }

}
